Made minor change to the gui