Minor changes to README and setup