Updates based on reviewer feedback

Minor updates to README for clarity and corrections to data/models setup script.