added code to repo, so can easily run it in recitation