Some fixes
Merge pull request #3 from Soorakh/master

Кое что подправил