added some ui code / not working yet