Improve categories import script