Minor: refactored output.
Rename to gradient demo.