Added test for main Class