Added couple of test cases and nailed this.