Update to account for new optional parameter