Refactored the UI to inherit from QGraphicsView