Adding tests for coverage