Small fix
Added comment