Added some tests in the main