Changed initialization, and added layers