Added the original python code I wrote