I made the model struct to store layers. I'm still missing the backward functions, but it's functional