Fixed the code to work again with coursera

Hi

I fixed to code so it works again with the coursera site