modifying demo code & readme