Allow a command callback to be set