Add wrapper to AFLW datasets