Add more tests. Add parameter to function to enable more specific tests and enables easier way to verify a valid function. Update problem description.
Add more tests. Add parameter to function to enable more specific tests and enables easier way to verify a valid function. Update problem description.