Forgot to do event listeners for button press