fixed example and updated doc string