Assignment 6.1

I think this works.