Tidied the code a little and updated the usage section.