fixed method and added test code
Added matplotlib example
Color coded graph
Added a normalizer function and a test for it