Close #102 Yes, it works. Also, added some more unit tests.