snake works
add button functionality