Add main (auto-test codes)