Finished pset2. Lotta fun