#21 Unit tests