Multi-function units for expansion tanks