add 2rd test case
refactor code
add 3rd test case
add readme
add 4th test case
refactor code