DOC: updated doc strings
MAINT: added tests