Formatting issue corrected to make code pretty