A little code cleanup. Updated the readme.