Added unit tests for all models