added docs
integrate comments