Added initial support to simulate the pump