Slight refactor
Added comment