Needed tests implemented. Need to find 2 more model classes and implement tests