Test code for python code works!