Minor changes to auto_encoders