Update module imports