Mark7549's picture
added docstring and removed unnecessary imported modules
c7d8395