Setup to handle both sets of tests
Added Comments explaining test scenario