Do some code clean