Updating generator for current_user