Adding python test file