added more functionality to gui