Improved testing suite. Doesn't work like REPL :(