This set of tests feels incomplete, any suggestions for additional test cases?