Fix example in README

Example wasn't working, so I modified as shown in the source's __main__ and also added some missing includes.