Tests

Adding some unit tests as beginning of my contributions.