US reinstates $2B in mental health, addiction funds after abrupt cuts