just a minor change to the function