Minor Fixes to the GUI

Minor changes to the GUI to make sure it is running the right methods
after each button.

Good job on the GUI though.