Updated category import