working on some ui stuff
working on ui