I should've added masks for accuracy