Implement tests (issue #9)