Added the tests to Main method