[pretty refactor] Simplify max_width API