Add unit tests for simple mode