Adding tests coverage