Small fix
Add comments