Updated tests to show coverage