diff --git a/README.md b/README.md
index 7755bad..a428c77 100644
--- a/README.md
+++ b/README.md
@@ -1,9 +1,9 @@
-# Deep Convolution Audio Generation (WIP)
+# Deep Convolution Audio Generation
 
-Implementing Deep Convolution to generate audio using generative and autoencoding networks
+Implementing Deep Convolution to generate audio using a generative network
 
 ## Directories
 
-- `model`: Model save and generated output location
-- `src`: Source code for models
+- `model`: Trained model and generated audio
+- `src`: Model source code
   - `utils`: Model and data utilities
diff --git a/src/dcgan_architecture.py b/src/architecture.py
similarity index 97%
rename from src/dcgan_architecture.py
rename to src/architecture.py
index 2328770..b01b9e4 100644
--- a/src/dcgan_architecture.py
+++ b/src/architecture.py
@@ -1,7 +1,5 @@
 import torch
 import torch.nn as nn
-
-# sys.path.append("../utils") # get this working & fix file structure
 from utils.helpers import N_CHANNELS, N_FRAMES, N_FREQ_BINS
 
 # Constants Constants
diff --git a/src/dcgan.py b/src/dcgan.py
index 8cf5d24..57e9655 100644
--- a/src/dcgan.py
+++ b/src/dcgan.py
@@ -3,12 +3,12 @@
 import torch.optim as optim
 from torch.utils.data import DataLoader, TensorDataset, random_split
 
-from dcgan_architecture import (
+from architecture import (
     BATCH_SIZE,
     Discriminator,
     Generator,
 )
-from dcgan_train import training_loop
+from train import training_loop
 from utils.helpers import (
     compiled_data_path,
     get_device,
diff --git a/src/dcgan_generate.py b/src/generate.py
similarity index 92%
rename from src/dcgan_generate.py
rename to src/generate.py
index 1556269..d3edcf4 100644
--- a/src/dcgan_generate.py
+++ b/src/generate.py
@@ -1,5 +1,5 @@
 import torch
-from dcgan_architecture import Generator, LATENT_DIM
+from architecture import Generator, LATENT_DIM
 from utils.helpers import amplitudes_to_wav, get_device, graph_spectrogram
 
 # Initialize Generator
diff --git a/src/dcgan_train.py b/src/train.py
similarity index 98%
rename from src/dcgan_train.py
rename to src/train.py
index 6436967..bfa7c23 100644
--- a/src/dcgan_train.py
+++ b/src/train.py
@@ -1,7 +1,6 @@
 import torch
-
+from architecture import LATENT_DIM
 from utils.helpers import save_model
-from dcgan_architecture import LATENT_DIM
 
 
 N_EPOCHS = 10
diff --git a/src/utils/helpers.py b/src/utils/helpers.py
index eee5a93..3687b9b 100644
--- a/src/utils/helpers.py
+++ b/src/utils/helpers.py
@@ -1,10 +1,10 @@
-import os
 import librosa
-import scipy
-import soundfile as sf
 import numpy as np
+import os
 import plotly.graph_objects as go
 import plotly.subplots as sp
+import scipy
+import soundfile as sf
 import torch
 
 # Constants