Improved the vote update callback to remove logic from controller