added styling bug fix