Adding button classes, support for extensions