add equals method back