Add mention of the header file to #import

I had to import the header to make it work