Added some testing and got a basis for the model