updates to tests and coverage