Add div for each class - need to style