Clean-up before script