replace Widget with Button in example