cleaned up some of the code, added basic ui placement in a hacky way