updated doc, allow python arrays as input