Implement a simple API for the Stage
Implement Stage#effective_count method
small changes to stage tests