Saving magic_square final code