Add tests

Added some testing.