Python Return 0 1