Add scale keyword argument.
Update demo.