Updated for pyside