Clean up of script