Added test to main and allowed test to be set