add a little improvement