diff --git a/.github/Architecture.md b/.github/Architecture.md
index ad96681e..e15429f3 100644
--- a/.github/Architecture.md
+++ b/.github/Architecture.md
@@ -5,220 +5,220 @@
 ```
 📦 backend
 |  |- 📂 common:
-|  |  |- 📜 optimizer.py : what optimizer to use (ie: SGD or Adam for now)
-|  |  |- 📜 ai_drive.py
-|  |  |- 📜 preprocessing.py
 |  |  |- 📜 default_datasets.py : store logic to load in default datasets from scikit-learn
-|  |  |- 📜 dataset.py : read in the dataset through URL or file upload
-|  |  |- 📜 email_notifier.py : Endpoint to send email notification of training results via API Gateway + AWS SES
-|  |  |- 📜 utils.py : utility functions that could be helpful
-|  |  |- 📜 constants.py : list of helpful constants
+|  |  |- 📜 ai_drive.py
+|  |  |- 📜 optimizer.py : what optimizer to use (ie: SGD or Adam for now)
 |  |  |- 📜 __init__.py
 |  |  |- 📜 kernel.py
+|  |  |- 📜 constants.py : list of helpful constants
+|  |  |- 📜 dataset.py : read in the dataset through URL or file upload
+|  |  |- 📜 preprocessing.py
+|  |  |- 📜 utils.py : utility functions that could be helpful
+|  |  |- 📜 email_notifier.py : Endpoint to send email notification of training results via API Gateway + AWS SES
 |  |  |- 📜 loss_functions.py : loss function enum
+|  |- 📂 aws_helpers:
+|  |  |- 📂 dynamo_db_utils:
+|  |  |  |- 📜 userprogress_db.py
+|  |  |  |- 📜 constants.py : list of helpful constants
+|  |  |  |- 📜 DynamoUnitTests.md
+|  |  |  |- 📜 trainspace_db.py
+|  |  |  |- 📜 dynamo_db_utils.py
+|  |  |- 📜 __init__.py
 |  |- 📂 ml:
 |  |  |- 📜 ml_trainer.py : train a classical machine learning learning model on the dataset
-|  |  |- 📜 __init__.py
 |  |  |- 📜 ml_model_parser.py
+|  |  |- 📜 __init__.py
 |  |- 📂 dl:
+|  |  |- 📜 dl_model_parser.py : parse the user specified pytorch model
 |  |  |- 📜 dl_model.py : torch model based on user specifications from drag and drop
+|  |  |- 📜 __init__.py
 |  |  |- 📜 dl_eval.py : Evaluation functions for deep learning models in Pytorch (eg: accuracy, loss, etc)
 |  |  |- 📜 dl_trainer.py : train a deep learning model on the dataset
 |  |  |- 📜 detection.py
-|  |  |- 📜 __init__.py
-|  |  |- 📜 dl_model_parser.py : parse the user specified pytorch model
-|  |- 📂 aws_helpers:
-|  |  |- 📂 dynamo_db_utils:
-|  |  |  |- 📜 trainspace_db.py
-|  |  |  |- 📜 DynamoUnitTests.md
-|  |  |  |- 📜 constants.py : list of helpful constants
-|  |  |  |- 📜 dynamo_db_utils.py
-|  |  |  |- 📜 userprogress_db.py
-|  |  |- 📜 __init__.py
+|  |- 📜 __init__.py
+|  |- 📜 data.csv : data csv file for use in the playground
+|  |- 📜 pyproject.toml
 |  |- 📜 middleware.py
 |  |- 📜 poetry.lock
-|  |- 📜 epoch_times.csv
-|  |- 📜 pyproject.toml
-|  |- 📜 data.csv : data csv file for use in the playground
-|  |- 📜 __init__.py
 |  |- 📜 app.py : run the backend (entrypoint script)
+|  |- 📜 epoch_times.csv
 ```
 
 ## Frontend Architecture
 
 ```
 📦 frontend
+|  |- 📂 layer_docs:
+|  |  |- 📜 Softmax.md : Doc for Softmax layer
+|  |  |- 📜 Linear.md : Doc for Linear layer
+|  |  |- 📜 ReLU.md : Doc for ReLU later
+|  |  |- 📜 softmax_equation.png : PNG file of Softmax equation
 |  |- 📂 public:
 |  |  |- 📂 images:
 |  |  |  |- 📂 wiki_images:
 |  |  |  |  |- 📜 tanh_plot.png
-|  |  |  |  |- 📜 avgpool_maxpool.gif
+|  |  |  |  |- 📜 dropout_diagram.png
+|  |  |  |  |- 📜 tanh_equation.png
 |  |  |  |  |- 📜 conv2d2.gif
+|  |  |  |  |- 📜 batchnorm_diagram.png
 |  |  |  |  |- 📜 conv2d.gif
-|  |  |  |  |- 📜 sigmoid_equation.png
 |  |  |  |  |- 📜 maxpool2d.gif
 |  |  |  |  |- 📜 softmax_equation.png : PNG file of Softmax equation
-|  |  |  |  |- 📜 batchnorm_diagram.png
-|  |  |  |  |- 📜 dropout_diagram.png
-|  |  |  |  |- 📜 tanh_equation.png
+|  |  |  |  |- 📜 sigmoid_equation.png
+|  |  |  |  |- 📜 avgpool_maxpool.gif
 |  |  |  |- 📂 learn_mod_images:
-|  |  |  |  |- 📜 neuron.png
-|  |  |  |  |- 📜 tanhactivation.png
-|  |  |  |  |- 📜 neuronWithEquation.png
-|  |  |  |  |- 📜 sigmoidactivation.png
+|  |  |  |  |- 📜 lossExample.png
+|  |  |  |  |- 📜 robotImage.jpg
+|  |  |  |  |- 📜 LeakyReLUactivation.png
+|  |  |  |  |- 📜 neuralnet.png
 |  |  |  |  |- 📜 lossExampleTable.png
+|  |  |  |  |- 📜 binarystepactivation.png
+|  |  |  |  |- 📜 ReLUactivation.png
 |  |  |  |  |- 📜 sigmoidfunction.png
+|  |  |  |  |- 📜 tanhactivation.png
+|  |  |  |  |- 📜 neuronWithEquation.png
+|  |  |  |  |- 📜 neuron.png
 |  |  |  |  |- 📜 lossExampleEquation.png
-|  |  |  |  |- 📜 neuralnet.png
-|  |  |  |  |- 📜 LeakyReLUactivation.png
-|  |  |  |  |- 📜 ReLUactivation.png
-|  |  |  |  |- 📜 robotImage.jpg
-|  |  |  |  |- 📜 binarystepactivation.png
-|  |  |  |  |- 📜 lossExample.png
+|  |  |  |  |- 📜 sigmoidactivation.png
 |  |  |  |- 📂 logos:
 |  |  |  |  |- 📂 dlp_branding:
-|  |  |  |  |  |- 📜 dlp-logo.svg : DLP Logo, duplicate of files in public, but essential as the frontend can't read public
 |  |  |  |  |  |- 📜 dlp-logo.png : DLP Logo, duplicate of files in public, but essential as the frontend can't read public
+|  |  |  |  |  |- 📜 dlp-logo.svg : DLP Logo, duplicate of files in public, but essential as the frontend can't read public
+|  |  |  |  |- 📜 dsgt-logo-light.png
+|  |  |  |  |- 📜 pandas-logo.png
+|  |  |  |  |- 📜 github.png
+|  |  |  |  |- 📜 flask-logo.png
 |  |  |  |  |- 📜 react-logo.png
-|  |  |  |  |- 📜 google.png
+|  |  |  |  |- 📜 aws-logo.png
 |  |  |  |  |- 📜 pytorch-logo.png
-|  |  |  |  |- 📜 pandas-logo.png
+|  |  |  |  |- 📜 dsgt-logo-white-back.png
 |  |  |  |  |- 📜 dsgt-logo-dark.png
-|  |  |  |  |- 📜 dsgt-logo-light.png
 |  |  |  |  |- 📜 python-logo.png
-|  |  |  |  |- 📜 dsgt-logo-white-back.png
-|  |  |  |  |- 📜 github.png
-|  |  |  |  |- 📜 aws-logo.png
-|  |  |  |  |- 📜 flask-logo.png
+|  |  |  |  |- 📜 google.png
 |  |  |  |- 📜 demo_video.gif : GIF tutorial of a simple classification training session
-|  |  |- 📜 robots.txt
-|  |  |- 📜 dlp-logo.ico : DLP Logo
-|  |  |- 📜 index.html : Base HTML file that will be initially rendered
 |  |  |- 📜 manifest.json : Default React file for choosing icon based on
-|  |- 📂 layer_docs:
-|  |  |- 📜 Linear.md : Doc for Linear layer
-|  |  |- 📜 ReLU.md : Doc for ReLU later
-|  |  |- 📜 softmax_equation.png : PNG file of Softmax equation
-|  |  |- 📜 Softmax.md : Doc for Softmax layer
+|  |  |- 📜 index.html : Base HTML file that will be initially rendered
+|  |  |- 📜 dlp-logo.ico : DLP Logo
+|  |  |- 📜 robots.txt
 |  |- 📂 src:
-|  |  |- 📂 pages:
-|  |  |  |- 📂 train:
-|  |  |  |  |- 📜 index.tsx
-|  |  |  |  |- 📜 [train_space_id].tsx
-|  |  |  |- 📜 _app.tsx
-|  |  |  |- 📜 _document.tsx
-|  |  |  |- 📜 LearnContent.tsx
-|  |  |  |- 📜 wiki.tsx
-|  |  |  |- 📜 learn.tsx
-|  |  |  |- 📜 forgot.tsx
-|  |  |  |- 📜 settings.tsx
-|  |  |  |- 📜 dashboard.tsx
-|  |  |  |- 📜 login.tsx
-|  |  |  |- 📜 about.tsx
-|  |  |  |- 📜 feedback.tsx
+|  |  |- 📂 backend_outputs:
+|  |  |  |- 📜 model.pt : Last model.pt output
+|  |  |  |- 📜 my_deep_learning_model.onnx : Last ONNX file output
+|  |  |  |- 📜 model.pkl
 |  |  |- 📂 common:
-|  |  |  |- 📂 styles:
-|  |  |  |  |- 📜 globals.css
-|  |  |  |  |- 📜 Home.module.css
 |  |  |  |- 📂 redux:
 |  |  |  |  |- 📜 userLogin.ts
+|  |  |  |  |- 📜 train.ts
 |  |  |  |  |- 📜 backendApi.ts
 |  |  |  |  |- 📜 store.ts
 |  |  |  |  |- 📜 hooks.ts
-|  |  |  |  |- 📜 train.ts
-|  |  |  |- 📂 utils:
-|  |  |  |  |- 📜 dndHelpers.ts
-|  |  |  |  |- 📜 firebase.ts
-|  |  |  |  |- 📜 dateFormat.ts
-|  |  |  |  |- 📜 nameFormat.ts
+|  |  |  |- 📂 styles:
+|  |  |  |  |- 📜 Home.module.css
+|  |  |  |  |- 📜 globals.css
 |  |  |  |- 📂 components:
 |  |  |  |  |- 📜 Spacer.tsx
 |  |  |  |  |- 📜 Footer.tsx
-|  |  |  |  |- 📜 HtmlTooltip.tsx
+|  |  |  |  |- 📜 TitleText.tsx
 |  |  |  |  |- 📜 DlpTooltip.tsx
-|  |  |  |  |- 📜 ClientOnlyPortal.tsx
-|  |  |  |  |- 📜 NavBarMain.tsx
+|  |  |  |  |- 📜 HtmlTooltip.tsx
 |  |  |  |  |- 📜 EmailInput.tsx
-|  |  |  |  |- 📜 TitleText.tsx
-|  |  |- 📂 backend_outputs:
-|  |  |  |- 📜 my_deep_learning_model.onnx : Last ONNX file output
-|  |  |  |- 📜 model.pt : Last model.pt output
-|  |  |  |- 📜 model.pkl
+|  |  |  |  |- 📜 NavBarMain.tsx
+|  |  |  |  |- 📜 ClientOnlyPortal.tsx
+|  |  |  |- 📂 utils:
+|  |  |  |  |- 📜 nameFormat.ts
+|  |  |  |  |- 📜 firebase.ts
+|  |  |  |  |- 📜 dndHelpers.ts
+|  |  |  |  |- 📜 dateFormat.ts
 |  |  |- 📂 features:
 |  |  |  |- 📂 Train:
+|  |  |  |  |- 📂 constants:
+|  |  |  |  |  |- 📜 trainConstants.ts
+|  |  |  |  |- 📂 types:
+|  |  |  |  |  |- 📜 trainTypes.ts
 |  |  |  |  |- 📂 redux:
 |  |  |  |  |  |- 📜 trainspaceSlice.ts
 |  |  |  |  |  |- 📜 trainspaceApi.ts
 |  |  |  |  |- 📂 features:
 |  |  |  |  |  |- 📂 Image:
-|  |  |  |  |  |  |- 📂 redux:
-|  |  |  |  |  |  |  |- 📜 imageActions.ts
-|  |  |  |  |  |  |  |- 📜 imageApi.ts
-|  |  |  |  |  |  |- 📂 types:
-|  |  |  |  |  |  |  |- 📜 imageTypes.ts
 |  |  |  |  |  |  |- 📂 constants:
 |  |  |  |  |  |  |  |- 📜 imageConstants.ts
+|  |  |  |  |  |  |- 📂 types:
+|  |  |  |  |  |  |  |- 📜 imageTypes.ts
+|  |  |  |  |  |  |- 📂 redux:
+|  |  |  |  |  |  |  |- 📜 imageApi.ts
+|  |  |  |  |  |  |  |- 📜 imageActions.ts
 |  |  |  |  |  |  |- 📂 components:
-|  |  |  |  |  |  |  |- 📜 ImageParametersStep.tsx
-|  |  |  |  |  |  |  |- 📜 ImageTrainspace.tsx
 |  |  |  |  |  |  |  |- 📜 ImageFlow.tsx
 |  |  |  |  |  |  |  |- 📜 ImageDatasetStep.tsx
+|  |  |  |  |  |  |  |- 📜 ImageTrainspace.tsx
 |  |  |  |  |  |  |  |- 📜 ImageReviewStep.tsx
+|  |  |  |  |  |  |  |- 📜 ImageParametersStep.tsx
 |  |  |  |  |  |  |- 📜 index.ts
 |  |  |  |  |  |- 📂 Tabular:
+|  |  |  |  |  |  |- 📂 constants:
+|  |  |  |  |  |  |  |- 📜 tabularConstants.ts
+|  |  |  |  |  |  |- 📂 types:
+|  |  |  |  |  |  |  |- 📜 tabularTypes.ts
 |  |  |  |  |  |  |- 📂 redux:
 |  |  |  |  |  |  |  |- 📜 tabularActions.ts
 |  |  |  |  |  |  |  |- 📜 tabularApi.ts
-|  |  |  |  |  |  |- 📂 types:
-|  |  |  |  |  |  |  |- 📜 tabularTypes.ts
-|  |  |  |  |  |  |- 📂 constants:
-|  |  |  |  |  |  |  |- 📜 tabularConstants.ts
 |  |  |  |  |  |  |- 📂 components:
 |  |  |  |  |  |  |  |- 📜 TabularParametersStep.tsx
-|  |  |  |  |  |  |  |- 📜 TabularTrainspace.tsx
 |  |  |  |  |  |  |  |- 📜 TabularDatasetStep.tsx
-|  |  |  |  |  |  |  |- 📜 TabularReviewStep.tsx
+|  |  |  |  |  |  |  |- 📜 TabularTrainspace.tsx
 |  |  |  |  |  |  |  |- 📜 TabularFlow.tsx
+|  |  |  |  |  |  |  |- 📜 TabularReviewStep.tsx
 |  |  |  |  |  |  |- 📜 index.ts
-|  |  |  |  |- 📂 types:
-|  |  |  |  |  |- 📜 trainTypes.ts
-|  |  |  |  |- 📂 constants:
-|  |  |  |  |  |- 📜 trainConstants.ts
 |  |  |  |  |- 📂 components:
 |  |  |  |  |  |- 📜 CreateTrainspace.tsx
-|  |  |  |  |  |- 📜 TrainspaceLayout.tsx
 |  |  |  |  |  |- 📜 DatasetStepLayout.tsx
-|  |  |  |- 📂 OpenAi:
-|  |  |  |  |- 📜 openAiUtils.ts
+|  |  |  |  |  |- 📜 TrainspaceLayout.tsx
+|  |  |  |- 📂 Feedback:
+|  |  |  |  |- 📂 redux:
+|  |  |  |  |  |- 📜 feedbackApi.ts
 |  |  |  |- 📂 Dashboard:
 |  |  |  |  |- 📂 redux:
 |  |  |  |  |  |- 📜 dashboardApi.ts
 |  |  |  |  |- 📂 components:
-|  |  |  |  |  |- 📜 TrainBarChart.tsx
 |  |  |  |  |  |- 📜 TrainDataGrid.tsx
+|  |  |  |  |  |- 📜 TrainBarChart.tsx
 |  |  |  |  |  |- 📜 TrainDoughnutChart.tsx
-|  |  |  |- 📂 Feedback:
-|  |  |  |  |- 📂 redux:
-|  |  |  |  |  |- 📜 feedbackApi.ts
 |  |  |  |- 📂 LearnMod:
-|  |  |  |  |- 📜 FRQuestion.tsx
-|  |  |  |  |- 📜 LearningModulesContent.tsx
-|  |  |  |  |- 📜 ModulesSideBar.tsx
 |  |  |  |  |- 📜 MCQuestion.tsx
-|  |  |  |  |- 📜 ClassCard.tsx
 |  |  |  |  |- 📜 ImageComponent.tsx
+|  |  |  |  |- 📜 ModulesSideBar.tsx
 |  |  |  |  |- 📜 Exercise.tsx
-|  |  |- 📜 next-env.d.ts
+|  |  |  |  |- 📜 FRQuestion.tsx
+|  |  |  |  |- 📜 LearningModulesContent.tsx
+|  |  |  |  |- 📜 ClassCard.tsx
+|  |  |  |- 📂 OpenAi:
+|  |  |  |  |- 📜 openAiUtils.ts
+|  |  |- 📂 pages:
+|  |  |  |- 📂 train:
+|  |  |  |  |- 📜 index.tsx
+|  |  |  |  |- 📜 [train_space_id].tsx
+|  |  |  |- 📜 login.tsx
+|  |  |  |- 📜 feedback.tsx
+|  |  |  |- 📜 learn.tsx
+|  |  |  |- 📜 dashboard.tsx
+|  |  |  |- 📜 about.tsx
+|  |  |  |- 📜 _document.tsx
+|  |  |  |- 📜 _app.tsx
+|  |  |  |- 📜 settings.tsx
+|  |  |  |- 📜 forgot.tsx
+|  |  |  |- 📜 LearnContent.tsx
+|  |  |  |- 📜 wiki.tsx
 |  |  |- 📜 iris.csv : Sample CSV data
 |  |  |- 📜 GlobalStyle.ts
 |  |  |- 📜 constants.ts
-|  |- 📜 next-env.d.ts
-|  |- 📜 tsconfig.json
+|  |  |- 📜 next-env.d.ts
 |  |- 📜 next.config.js
 |  |- 📜 jest.config.js
 |  |- 📜 .eslintrc.json
-|  |- 📜 pnpm-lock.yaml
 |  |- 📜 package.json
+|  |- 📜 tsconfig.json
+|  |- 📜 pnpm-lock.yaml
+|  |- 📜 next-env.d.ts
 |  |- 📜 .eslintignore
 ```