assigning classes fix