update comments and docs