Remving main method from tests