Changes in the if statement. Buttons are now shown on the screen