Tests

Added tests for models.