adding initial python code
adding python