Implement `wait for a number of calls` feature (Fixes #81)