I have a qualm with the sample test cases provided. Don't use 0 or 1, because most of the times they represent some boundary case or special case.

Maybe you can include one example with a bigger number, like 4 or 5?