Add method to User model