refining logic in comments