Merge pull request #13 from AlmogCohen/develop

Added missing import for InvalidArgumentError