still working on getting the UI fully working