updated tests, need to add some more