Fixing issue #3

Yes, thank you