Add simple unit tests for everything that involves logic.
Add simple unit tests for everything that involves logic.