Doc improvements

Updated docs