Add code adding and getting categories