fixed issue #2

used the example from the link in issue #2  to create this quick fix.