few fixes on styling