refactor/ add more tests