Add the possibility to use dict of lists in the map function
updated doc string