Update schema to include collections