updates to auto encoder