Added unit tests to features implemented in v0.1.0