Option to set callback directly