Minor styling changes to demo