Update import section

Now reflects the new simple import