Python script to solve a simple electrostatics problem numerically