Working on revising the autoencoder into layers.