Fix update check
Updating UI
Merge pull request #332 from codewithpassion/fix-update-function

Fix update function