Updated Main function of PumpController test code.