Python Code

Added the start of the python code.