Add some missing comments