Added some comments to clarify what I'm trying to do. I need to figure out generators and put it in a class.