More refactoring (Karthik)