Revise UI of demo