Improve snake update method