Budget lays down plan to invest in infrastructure