Fixes #9 by adding string input