Add SampleWidget class