Setting up testing functionality