Clean up script

Clean up