Slight change in the logic