Small fix
Added comments