Add remove_button option. Remove all the debug output. Closes #10