Fix styling

Some minor problems existed with the styling in the last version.
Those have been resolved.