remove unnecessary classes
use forEach instead of for loop