Added unit tests using mock
Merge pull request #36 from trobert2/all_tested_with_mock

Added unit tests using mock