Added new feature
I fixed the thing commented in pull request