Improve current_user method to set instance variable