Add python version and unit test