Add simple logic and tests.