Correction of HW6.ipynb