Fixes #4: Add simple unit testing