Add code to remove temp file