Fixed error, improved demo.