Add test for User class