WIP: making improvements in docs