added treatmet logic