Made the example work with my own input