Updated schema migration