Fixed a couple of errors in the source code. Funny how they got through :/