refactor buttons to one class