added missing classes :)