added a two-layer network that gets higher accuracy