Code cleanup for the new snake movement