Re-working snake example