hw2.ml done, need more tests still