Different shapes for looking around the square might be possible. * Maybe use a mask ? In tensorflow or in python * For now the game state is 27 number, and it is **hard-coded**, which is bad for modularity... We might have states of various sizes