improve code style fixed #5