I've made some changes

pull request