Updates to allow for intercept term in AFT