Minor fiddling with the autoencoder