added missing class def