Fixing issue#9

Please add this. 

Thanks