Small update
Add comments