Small changes made to GUI