Adding .py extension to script automatically