Minor changes to rolloutpolicy function