Cleaned up script output when done.

Fixes #4