reduce should return a tuple
Add tests for class Stage