Change to new update functionality
Merge pull request #327 from codewithpassion/fix-update-function

Change to new update functionality