fix styling after design review