Add .py extension