remove output and add comments