Import only the needed modules