Update schema and category predictions