small changes

A couple of last min changes that I have been working on.