Can you please remove the **trailing white spaces** at the end of lines here and add a **doc string** to the new class? :)