added function for WB dataset cleaning