Adding to Widget class