Computational Physics 301, Exercise 2