Add tests for user model