Adds basic test for main class. More to come