Add styles
Small refactoring