Fix references in Andrew's assignment 1