Minor changes
Cleaned the code a little bit.

No new features added