Some improvements :bust_in_silhouette:
Merge pull request #1 from piecioshka/master

A little improvements