Updated interface based on @peastman suggestions. Started to convert docstrings to numpy format. Updated test with some small PDBs.
