Update + improvements

Hi there,

Thank you for sharing your code. I tweaked it a little bit to allow a different MD engine to work. I also fixed small things and increased the version number.

Could you merge my code?

Thanks and keep up excellent work.