Clean up script formatting