Adding tests
Version bump (v.0.4.1)