added the python file and a relevant data text file