remove main to test