Adding support for getting style from button