code cleaning

code cleaning