Add Tests

Fixes #1