Add comments to button class