minor modification for PEP8