remake styling a bit