added test
Automatic style corrections courtesy of autopep8