Add tests

Adds a test suite :) Please review @perry