fixed styles after refactoring