improved the GUI