Add tests for User model