adding .py to the code as file extension