cleaning the code
improvising the code