Add migration for plan_id