Minor update

* styling