minor fixes. Take a look at coverage gaps