updated doc string, removed units