cleaning the code
fix