Improvement of the code.

- Changing tabs to two spaces.
- Add comments