Fixed imports to reflect code

I have changed the imports to reflect the code.  Still v. new to git so I hope I have gotten this right