added a button feature