code cleaning

cleaning code