Virgin Racing's technical director Nick Wirth has promised a major upgrade on the VR-01 for the Bahrain Grand Prix