updated readme, improved demo code