Thanks for improvements

Hi michael, thanks for approving my pull request. I'll try your code