Update cleanup.rb

Tried to implement it with (kind of) one-line-of-code...