Slight modifications GUI.cs