Formatting the code properly