added more UI elements, working on making them look nicer