removed print statment
updated input
renamed to demo.py