Add some more conditonal styling