Solved the TODO question, but still I do have a problem executing the last instructions of the notebook.

Am I supposed to solve any other bug hidden in the code?
Thanks for you help.