almost done the gui part