added tests and changes models.pybecause tests not working