[Refactor the code]