Added import in code. Was causing issues
Merge pull request #20 from treadstone90/upstream

Added import in code. Was causing issues