arrange source code

arrange source code