add some styling and minor changes