Upgrading the pump system