add main class
Merge pull request #2 from HaiNNT/master

We need a diagram for these.