Updated auto-encoder test