assignment 3: add comments