add unit tests for the models