some changes

Just corrected some lines