Added doc to code