Modified the spacing in the code to meet the PEP style