Modification to GUI