PEP8
Renaming decorator and removing double import in test file.
Merge pull request #1 from bronikowski/master

A tiny cleanup