Proper showing of subpackages

NOT YET TESTED
Better visualisation and now tested

Also added a legend
Merge pull request #34 from soosterwaal/issue_30

Issue 30