Minor changes to calls in script