Small updates to style