Add basic commenting to code