Fixed one error so it all works together now.
Merge pull request #9 from alexe0131/master

Fixed one error so it all works together now.