add back missed classes