clean up the code added test