Styling errors fix