updated tests to include coverage