updated tests
added coverage