Code formatting
Merge pull request #3 from achwedyk/patch-1

Code formatting in tests