Update for Puzzle and arrow trap prefabs