Switched to ternary operator for if statement
Changed add and remove buttons to primary and danger, respectively