Medium Python Code For Snake