Merge pull request #19 from treadstone90/upstream

Added Thats what she sad replier to tshrdlu
Added import in code. Was causing issues