Made small changes to GUI