Change if current_user to if signed_in?