Button class update