adds more test case and the solution
adds comments in the solution