Number pyramid

Here is the code for building up a number pyramid where the line goes from n^2+1 up to (n+1)^2

and the output looks like this (primes are denoted by yellow ink):

Code is available at https://github.com/CAG12/python/blob/main/number_pyramid.ipynb