Add tests to distribution