update docs
small code move