Eddited code to meet PEP8 standards