Adding changes to autoencoder section.