Improve styles for demo
Update demo for key change