Add User class for testing