add styles refactor