Clean up script a little