added some changes

clear :)