Немного "причесал" код )