Added sample tests

Closes #1.