Correted part 2 from Homework 0