Add tests

Add initial testing support.