HW3 update

Added some quick tests to make sure my functions were working right.