Add test for User model