Updated class doc string to be more clear