Changed input to number
Merge pull request #6 from marcuspoehls/issue-#2

resolved issue #2 and #3