fix imports and add missing button styles