made some small changes, started UI