updated tests and added an edge test with trickier cases