Updated database schema with migration