Added more coverage