Add README and code cleanup