3rd year Computational Physics assignment.