DOC: Add module level doc
STY: PEP8-ify