Clean up and update readme