Cleaning the code.
Adding comments.