Changes for Python plotting