first attempt at having an input for the button