Small changes

But I finally figured out what I am going to do!