Add comments for documentation