improve on docs