Updated outdated class names.