add tests to ensure puzzle method works correctly