Add large sample. Rewrite python to work with both large sample and small sample. Update python benchmark.