basic version
accepted sol
add test cases with another slower sol
add comment