added some imports to tests and not sure of initialising as two classes for diff and patch