added UI stuff