Unit Test for main classes