Added tests for what to do for puzzle box