Variable Input Python Code For Snake