updated unit tests for existing models