Added functionality to the GUI2