Fix the tests
Tweak documentation
Start breaking the code into smaller files.