Update clean-up script

Signed-off-by: Kris Hicks <krishicks@gmail.com>