Fix some code forces problem

fix some code forces problem