Erased two mistakes from the code