adding some of the UI stuff