Clean up script. Not actual changes in code