Add tests

Tests!