Changed remove to a button.