Added remove button if logged in