I would drop the last baffle myself to make it a wet dry. The plant in there looks plastic, probably just there to show what you can do with the space. I wouldn't add more bio-balls after the box, the refugium will take care of many more problems then the added bio-ball, or even pot scrubbers.