unit test update + styling fix