Adding python code for demo app