Finally fixed it. Cleaned up the code a bit. Will release 0.1 when I finished docs and tests.