Add missing comments