Updated unit test references and names to corrispond to new class names.
Updated version to 0.5.4.