Added code to save pdb files