improve code efficiency
Merge pull request #2 from sunaiwen/improve_efficiency

improve code efficiency