modification for the gui