now fix styling