Add change status logic