[WIP] Support for custom styles