refactor for styling