Unit tests

Added the first set of unit tests.
They are pretty basic, but should be enough for a relatively
easy program.

[Close #5]