adding more tests
adding main function