code redesign

redesign