A plan to improve the productivity of UK businesses has been published by the Treasury.