Completed the homework for Class 4