made some minor changes to the GUI and the functionality