Import necessary module only