Added socket tracking

Attempts to fix issue #22
Better docs [MINOR]