Update for recent changes in code