As asked by lcv232 I'm pushing the problem3.py