Fixed the code section