updated the cleanup method