Python Computing Snake