Added initial python code