0.0.2 Add tests