Reduce Array Size Python Code For Snake