Small changes

Just a couple of small changes.