کد رو ویرایش کردم، در قسمت های مشخص شده قرار بدید.