add unit tests for models