add Test to main