forgot to add company class