Update import to new script name
PEP-8 cleanup