missed importing reshape library
Merge pull request #3 from nikita-barsukov/master

Added missing statement that imports reshape library