fix the styling