Simple example of creating a Ulam Spiral in Python