added a few tests in main