Updated the neatness of the source code.