added class fix