update comment and update wrapper to include the two new arguments