Small change to the GUI