Made a few tweaks to the GUI