Updated button class
Added button class