[Lab01]: Changes after code review.