Adds cleanup for heist mode