Fixed the formating of code