fix a bit of styling