Started writing test suite in main