Update styling to be nicer!