Tests added

For #39