Adding UIWidget and UIButton class