code edited

some changes