Should also remove the button