Wrote tests to get coverage back to standards