done with writeup, just need the python