Simple snake clone in pygame