removing deprecation warning, PEP8