Small touch ups to the code and readme.