Fixed a couple of small errors