More functionality added to GUI