More stuff to write tests for added
added more tests WKB