Moved python code to the code directory