improved tests. I love go :P