Add switch for cleaning up files