Add input for snake