Updated method to test