Computational physics homework