Adding first version of python script