Add visual changes to the GUI class