Revert "Revert "Added schema update""

This reverts commit 5a3d9be78699f3511e844041aa2c205ae74daf5f.