Correct logic for track 2