added coverage to code