Updated the code styling

Updated the code styling