Work with categories... maybe I should just add in a db?