Update styling a bit more