Change classes
Added button styling
Fixed button styling