Updates tests, now looks much better with regards to code coverage