Give the option to add a different css style

Tiny changes to accept style as an option.