Remove all the tests and polish the code a little.