Added the model
Finished implementation should now be working