adding in the tests I'll write tomorrow to build the model