Added README and code cleanup