Improve plotting example

Use step to plot, gives it the proper histogram look