fixed some missing classes