[fix] #1 - using class names instead of elements