update styling to match demo