Forgot to do event listeners for button press

Forgot one method