Made updates to GUI (polish)