Fix method in the second task