Fixed some code in demo