Initial upload of python code