Fixed test file
Added comments to make code clearer