Fix: add comparison methods