Fixing import file and function
Trying out some test cases
Trying out some test cases