Minor code improvements
Fixed example shader code