Would be good to add the dosctrings at the top of the file, i. e. __author__, __email__, etc.

It's not necessarily bad but there is only one comment :warning: