Push new changes

I've improved the code.
This pull request is to update your fork.