refactored pymol a bit
added docstrings to pymol