Fix on the fourth task in the first homework