Modified ga code to run on a toy magic square example