Move python code to its own branch