Only import what's necessary.
Merge pull request #7 from Raydiation/only_import_whats_necessary

Only import what's necessary.