Adding some code to cover