Style fix in button class