Fixing missing update statements for new functionality

Fixing missing update statements for new functionality