Changed some styling to be a bit better