added missing classes

sorry...