just a little modification ^_^