Minor change to GUI