Added some tests. This would serve as documentation to some extent.