Work on potential calculation example on notebook