added testing in python