Fix spaces in code

Light editing for spaces and comment in the code.