Add return method to User class