Minor style fixes to align elements properly