[#2] Add tests.