added some python code that I knew the answer to