The hook approach would likely work, a custom dataset may also