added tests to cover missed lines
changed one line for pull request
Merge pull request #72 from osulp/feature/MoreTests

Feature/more tests