Adding coverage tests