clean up litle bit