Updated comments to use the correct imports
Merge pull request #2 from svisser/patch-2

Updated comments to use the correct imports