Update GameOfLife.py

use list comprehension for the neighbour coords