Added check for current_user .