Replace ugly example with unit tests.