Added missing classes to Repository