Small change to GUI