Computational Physics hw 2