fix magic square example output