implement button press to quit and improve ui