Add code according to the requirement