Code works for add and sub