baby's first test
refactored and first tests added
Merge pull request #3 from angus-c/tests

Tests