Auto import statement add

Adding import statements for all classes that could be used by the
script