Working on the gui