clean the code

clean the code