added functionality to remove button. All currently existing buttons
work now!