How about just rock for the stream?

In the past couple days, I was talking to a friend about how I could have made my first vivarium better (I'm already tempted to dismantle and start over again. ), and I mentioned a waterfall and stream, and though I haven't put -that- much thought into it, I thought just having a nice thick layers of rock that would create just a shallow area for water would work. Maybe.