ZigZag problem. Add unit tests for expected outputs before implementation.