adding more tests
adding more tests
adding more tests
adding more tests tidy up the code