pep8
cleaned up decorator a little bit