[main.js] Add "snake" function