removing .py from arguments