Added button class as attr