#77 Unit tests