hw 0 adding lecture code