Can u check my logic in operation() function?