Add parameter for callback