Improve styling a bit