Simple snake python and pygame