fix styling and add a test