A little bit update

Update with new API