KSM[FIX]: Added update method