Updated schema with foreign key