Cleaned the code a bit (indentation and spacings).