Use smaller system
Implement rhs assembly
Solve the 2nd Poisson equation
Add debug prints
Get rid of an if statement
Optimize loops a bit
Add more printing statements