v0.0.4

Tests and refactoring of the class.