working on getting the UI to work