Add normalization function

[gitosaurus]

New normalize function takes any generally linear numpy array and returns a version suitable for visualization.

Also use if __name__ == '__main__' so that this can be used as a library module as well as a script.