Minor update

* minor changes