Playing with the Ulam spiral in python