Solution: Fix the Code

In this lesson, we will see how the code given in the previous lesson can be fixed to give the right output in every use case.