just clean the code litellbit