Add source term feature