added more to the UI