Add ID to model