Changes in style of the code

Now it is more readable