Clean up the repo, updates to README