Added functionality to GUI