Modified README to fit code changes