added tests

closes #4