Added new file with python code