updated class naming

inline with frontend docs