Added a button class
edited a comment