Adding documentation and making small behavior change
Adding documentation and making small behavior change