Add test for #8
Fix #8: IndexError on PLT
Merge pull request #9 from eliangcs/issues/8

Issues/8