replace Widget with Button in example (continued)