I've separated the "main" code initially, and also changed the PDB reade...

Still a work in progress, just filing pull request now. I've separated the __main__ code to another file, and changed the pdb reader to use openmm.app.