Add categories functionality