Treasury officials also played a key role on the federal budget announced last month and are working on an initiative to tackle the toxic assets clogging the books of banks as well as a plan to overhaul financial regulation .