Budget 'stage two' of virus recovery: PM