added UI to demo