Refactored for loop and ran  on both files
Merge pull request #2 from a-temlyakov/master

Refactored for loop and ran 'fmt' on both files