Add return_stats option to ta_{mb,}wrap