Fix minor bug, update example