Add unit tests

Closes #42
indentation (1 column too far out)