Modify(less code)