wrote tests for models