Code cleanup and new readme.