fixing the button

still in progress