small change to interface of SimpleAutoEncoder