Красиво оформил код.

Доводка кода.