Fix button styling

Fixes issue #7.