Python Difference Single Stage